package com.ctvit.cch.ai.service.impl;

import com.alibaba.fastjson2.JSON;
import com.ctvit.cch.ai.pojo.dto.AutoCategoryDTO;
import com.ctvit.cch.ai.pojo.dto.CreativeEvaluationDTO;
import com.ctvit.cch.ai.pojo.vo.AutoCategoryVO;
import com.ctvit.cch.ai.pojo.vo.CreativeEvaluationVO;
import com.ctvit.cch.ai.service.INlpService;
import dev.langchain4j.data.message.ChatMessage;
import dev.langchain4j.data.message.SystemMessage;
import dev.langchain4j.data.message.UserMessage;
import dev.langchain4j.model.chat.response.ChatResponse;
import dev.langchain4j.model.openai.OpenAiChatModel;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;

/**
 * 自然语言处理服务实现类
 * <p>
 * 该类实现了自然语言处理相关的业务逻辑，主要用于文本分类等NLP功能
 *
 * @author heyingcheng
 * @date 2025/07/30
 */
@Service
public class INlpServiceImpl implements INlpService {

    private final OpenAiChatModel siliconChatModel;

    public INlpServiceImpl(OpenAiChatModel siliconChatModel) {
        this.siliconChatModel = siliconChatModel;
    }

    /**
     * 自动分类方法
     * <p>
     * 根据输入的文本内容，使用自然语言处理技术自动对其进行分类
     *
     * @param dto 包含待分类文本信息的数据传输对象
     * @return AutoCategoryVO 分类结果的视图对象，包含分类标签和置信度等信息
     */
    @Override
    public AutoCategoryVO categorize(AutoCategoryDTO dto) {
        List<ChatMessage> messages = new ArrayList<>();
        messages.add(SystemMessage.from(AutoCategoryDTO.SYSTEM_MESSAGE_TEXT));
        messages.add(UserMessage.from(AutoCategoryDTO.USER_MESSAGE_TEMPLATE.formatted(StringUtils.defaultString(dto.getTitle()), StringUtils.defaultString(dto.getSummary()))));

        ChatResponse chatResponse = siliconChatModel.chat(messages);
        String categoryJsonText = chatResponse.aiMessage().text();
        return JSON.parseObject(categoryJsonText, AutoCategoryVO.class);
    }

    /**
     * 该方法用于调用AI模型对创意内容进行评估。首先构建系统提示和用户消息，然后通过siliconChatModel发送请求并获取AI回复的JSON文本，最后将该JSON解析为CreativeEvaluationVO对象返回。
     *
     * @param dto 创意评估数据传输对象
     * @return {@link CreativeEvaluationVO } 创意评估结果对象
     */
    @Override
    public CreativeEvaluationVO evaluate(CreativeEvaluationDTO dto) {
        List<ChatMessage> messages = new ArrayList<>();
        messages.add(SystemMessage.from(CreativeEvaluationDTO.SYSTEM_MESSAGE_TEXT));
        messages.add(UserMessage.from(CreativeEvaluationDTO.USER_MESSAGE_TEMPLATE.formatted(StringUtils.defaultString(dto.getTitle()), StringUtils.defaultString(dto.getSummary()))));

        ChatResponse chatResponse = siliconChatModel.chat(messages);
        String evaluationJsonText = chatResponse.aiMessage().text();
        return JSON.parseObject(evaluationJsonText, CreativeEvaluationVO.class);
    }

}
